import datetime
import os.path


# 工具类
class Tools:

    # 保持截图的方法
    # driver是浏览器对象
    # error_type是错误类型字符串
    @classmethod
    def save_screenshot(cls, driver, error_type):
        pic_dir = './error_pics/'  # 图片保存的文件夹
        today_dir = datetime.datetime.now().strftime('%Y_%m_%d')  # 今天的日期
        dir_path = pic_dir + today_dir + '/'  # 今天的截屏保存路径
        if os.path.isdir(dir_path) == False:
            print('今天要保存的文件夹不存在')
            os.mkdir(dir_path)  # 创建一个文件夹
        # 通过当前时间生成一个文件名
        stime = datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
        # 文件完整路径名：图片所在文件夹/错误类型、日期时间.png
        f_name = f'{dir_path}{error_type}{stime}.png'
        # 截屏并保存
        driver.get_screenshot_as_file(f_name)


if __name__ == '__main__':
    from selenium import webdriver
    dr = webdriver.Chrome()
    dr.get("https://www.baidu.com")
    Tools.save_screenshot(dr, 'cuowu')
